package com.example.marksixanim

import junit.framework.Assert.assertEquals
import org.junit.Before
import org.junit.Test

/**
 * @author robert
 * @date 2017/12/25
 */
class BallInfoTest {

    lateinit var ballInfo: BallInfo
    @Before
    fun init() {
        ballInfo = BallInfo("3", "红", "鼠")
    }

    @Test
    fun getColorDrawable() {
        assertEquals("球对应的颜色", ballInfo.getColorDrawable(), R.drawable.bg_mark_six_ball_color_red)
    }

    @Test
    fun getAnimalDrawable() {
        assertEquals("球对应的动物", ballInfo.getAnimalDrawable(), R.drawable.bg_mark_six_ball_animal_mouse)
    }

    @Test
    fun getNumber() {
        assertEquals("球对应的编号", ballInfo.getNumberDrawable(), R.drawable.bg_mark_six_red_number2)
    }
}

